package com.game.mp;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * @author lvhao
 * @date 2020/7/24 18:30
 * @description 代码生成器
 */
public class MpGenerator {
    public static void main(String[] args) {
        AutoGenerator autoGenerator = new AutoGenerator();

        // 设置数据源
        autoGenerator.setDataSource(new DataSourceConfig()
                        .setDriverName("com.mysql.cj.jdbc.Driver")
                        // 设置数据库类型
                        .setDbType(DbType.MYSQL)
                        .setUsername("root")
                        .setPassword("12345678")
                        .setUrl("jdbc:mysql://127.0.0.1:3306/game_background_data?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai")
        );

        // 全局配置
        autoGenerator.setGlobalConfig(new GlobalConfig()
                // 输出目录
                .setOutputDir("C:\\Users\\0.0\\Desktop\\游戏后台项目资料和源码\\源码\\my-game\\src\\main\\java")
                // 是否覆盖
                .setFileOverride(false)
                // 开启AR模式
                .setActiveRecord(true)
                // XML二级缓存
                .setEnableCache(false)
                // 生成ResultMap
                .setBaseResultMap(true)
                // 生成 sql片段
                .setBaseColumnList(true)
                // 自动打开生成后的文件夹
                .setOpen(true)
                // 所有文件的生成者
                .setAuthor("lvhao")
                // 自定义文件命名,%s会自动填充表实体类名字
                .setMapperName("%sDao")
                .setXmlName("%sMapper")
                .setServiceName("%sService")
                .setServiceImplName("%sServiceImpl")
                .setControllerName("%sController")
        );

        // 策略配置
        autoGenerator.setStrategy(new StrategyConfig()
                // 需要生成的表
                //.setInclude("email","intruder_log","intruder_data")
                // 实体类使用Lombok
                .setEntityLombokModel(true)
                // 表名生成策略,下划线转驼峰
                .setNaming(NamingStrategy.underline_to_camel)
        );

        // 包配置
        autoGenerator.setPackageInfo(new PackageConfig()
                // 基本包路径
                .setParent("com.game")
                // 设置Controller包名
                .setController("controller")
                // 设置entity包名
                .setEntity("pojo.po")
                // 设置Mapper包名
                .setMapper("dao")
                // 设置Service包名
                .setService("service")
                // 设置Service实现类包名
                .setServiceImpl("service.impl")
                // 设置Mapper.xml包名
                .setXml("mapper")
        );

        // 注入自定义配置
        autoGenerator.setCfg(new InjectionConfig() {
            @Override
            public void initMap() {
                Map<String, Object> map = new HashMap<>(1);
                map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-mp");
                this.setMap(map);
            }
        }.setFileOutConfigList(Collections.singletonList(
                new FileOutConfig("/templates/mapper.xml.vm") {
                    // 自定义Mapper.xml输出路径
                    @Override
                    public String outputFile(TableInfo tableInfo) {
                        return "C:\\Users\\0.0\\Desktop\\游戏后台项目资料和源码\\源码\\my-game\\src\\main\\resources\\mapper\\" + tableInfo.getEntityName() + "Mapper.xml";
                    }
                })));

        autoGenerator.execute();
    }
}
